ubuntu安装并切换多个版本的cuda |
您所在的位置：网站首页 › ubuntu cuda10 › ubuntu安装并切换多个版本的cuda |
ubuntu安装并切换多个版本的cuda
|
1.背景
租用的服务器，里面安装的cuda版本是cuda10.1，需要配套的tensorflow-gpu在1.13.1以上，我之前用的是cuda9.0，cudnn7.6.0，tensorflow-gpu=1.8 不想再修改实验代码，所以就打算安装多版本的cuda 2.安装先去cuda官网下载需要的cuda版本https://developer.nvidia.com/cuda-toolkit-archive 进入到cuda_9.0.176_384.81_linux.run文件所在的目录中（前面的cuda_9.0.176代表cuda的版本，后面的384.81代表的对应的NVIDIA驱动的版本） 先执行下面的命令安装相关依赖，否则会出现Missing recommended library错误 sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev 1开始安装 sudo sh cuda_9.0.176_384.81_linux.run 1 #直接按q退出协议说明. accept/decline/quit: accept #接受协议 Install NVIDIA Accelerated Graphics Driver...? y)es/(n)o/(q)uit: n #已经安装显卡驱动，选择n Install the CUDA 9.0 Toolkit? (y)es/(n)o/(q)uit: y #是否安装工具包，选择y Enter Toolkit Location [ default is /usr/local/cuda-9.0 ]: #工具包安装地址，默认回车即可 Do you want to install a symbolic link at /usr/local/cuda? (y)es/(n)o/(q)uit: y #添加链接注意这个连接，如果你不想使用之前安装过的另一个版本的cuda #要使用新版本的，就选择y，否则这里就建议选n，因为指定该链接后会将cuda指向这个新的版本 Install the CUDA 9.0 Samples? (y)es/(n)o/(q)uit: y #安装样例 Enter CUDA Samples Location [ default is /root ]: #样例安装地址默认即可 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21配置cuda的环境变量，修改vim ~/.bashrc文件，在末尾添加 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 export PATH=$PATH:/usr/local/cuda/bin export CUDA_HOME=$CUDA_HOME:/usr/local/cuda 1 2 3使用 source ~/.bashrc 1使得配置生效 3.多个版本切换可以在/usr/local/目录下查看自己安装的cuda版本 cd /usr/local/ ls 1 2
我们安装了cuda10.1，cuda9.0，而cuda是一个软链接，它指向我们指定的cuda版本 在设置环境变量时，使用的是cuda，而不是cuda-10.1和cuda-9.0，主要是方便我们切换cuda版本，不用每次都去设置环境变量的值 stat cuda 1查看cuda，发现文件类型是symbolic link，而指向的目录正是/usr/local/cuda-9.0，当我们想使用cuda-10.1版本时，只需要删除该软链接，然后重新建立指向cuda-10.1版本的软链接即可 sudo rm -rf cuda sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda |
我下载的是cuda_9.0.176_384.81_linux.run
【本文地址】